使用Gorm动态更新数据表中的字段

2024-07-11 01:52| 来源: 网络整理| 查看: 265

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

文章目录前言一、pandas是什么？二、使用步骤1.实现方式

前言

使用Gorm动态更新数据表中的字段：

当使用API 更新数据表中的记录时，会存在只需要更新部分字段的需求。

这里我使用了Gin 构建API, 通过Gorm进行数据表的操作

假设我们有如下的API 以及对应model的struct：

一、pandas是什么？ //API PUT /users //Body { "id": 16, "realName": "Tester1" } 二、使用步骤

代码如下（示例）：

type SysUser struct { ID int64 `gorm:"column:id;primaryKey;autoIncrement:true" json:"id"` // id Username string `gorm:"column:username;not null" json:"username"` // 用户名 Password string `gorm:"column:password" json:"password"` // 密码 RealName string `gorm:"column:real_name" json:"real_name"` // 真名 Email string `gorm:"column:email" json:"email"` // 邮箱 Mobile string `gorm:"column:mobile" json:"mobile"` // 手机号 1.实现方式

实现方式

这里我用map[string]interface{} 来接收request body 中的数据, 在接受数据之后再进行json key 和 gorm column key 的转换。

map[string]interface{} 接收request body

这里使用io.ReadAll 读取context 中的body数据，之后使用json.Unmarshal 把context body中读取到的数据转换成map[string]interface{}

json key 转换 gorm column key

这里使用reflect 获取struct 中的fields, 使用filed 中的Tag 获取json key，然后使用Gorm中的schema.ParseTagSetting 获取column key

主要的技术细节如上，详细的可以参考code部分

Code

data = pd.read_csv( 'https://labfile.oss.aliyuncs.com/courses/1283/adult.data.csv') print(data.head())

该处使用的url网络请求的数据。

func main(){ router := gin.New() router.PUT("/users", func(ctx *gin.Context) { m, err := common.GetReqBodyAsMap(ctx) if err != nil { logger.Err(err.Error(), ctx, err) ctx.JSON(http.StatusBadrequest, "Fail") return } err = UpdateUser(ctx, m) if err != nil { logger.Err(err.Error(), ctx, err) ctx.JSON(http.StatusBadrequest, "Fail") return } ctx.JSON(http.StatusOK, "Success") }) } router.Run(":8080") } func GetReqBodyAsMap(ctx *gin.Context) (map[string]interface{}, error) { bodyData, _ := io.ReadAll(ctx.Request.Body) data := make(map[string]interface{}) err := json.Unmarshal(bodyData, &data) if err != nil { return nil, err } return data, nil } func ConvertToGormKey(m map[string]interface{}, obj any) { t := reflect.TypeOf(obj) for i := 0; i delete(m, jsonKey) m[tagSetting["COLUMN"]] = reflect.ValueOf(val).Convert(sf.Type).Interface() } } } func UpdateUser(ctx *gin.Context, m map[string]interface{}) err { ConvertToGormKey(m, SysUser{}) su := dao.Use(dao.Db()).SysUser rstInfo, err := su.WithContext(ctx).Where(su.ID.Eq(m["id"].(int64))).Updates(&m) if rstInfo.RowsAffected == 0 { err = errors.New("no rows were updated") logger.Err("Update failed", ctx, err) return err } return err }

【本文地址】

公司简介

联系我们